Home
last modified time | relevance | path

Searched refs:x_loc (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h116 llvm::Value* x_loc, int64 x_iter_num)>;
259 llvm::Value* y_loc, llvm::Value* x_loc,
266 llvm::Value* y_loc, llvm::Value* x_loc,
273 llvm::Value* y_loc, llvm::Value* x_loc,
Dir_emitter_unnested.cc97 llvm::Value* x_loc, int64 x_iter_num)>;
2345 llvm::Value* x_loc = builder->CreateAdd( in EmitFullElementalTile() local
2348 emit_elem_function(source_idx_y_x, y_loc, x_loc, j); in EmitFullElementalTile()
2376 llvm::Value* x_loc = builder->CreateAdd( in EmitPartialElementalTile() local
2380 loop_name + "_x_in_tile", builder->CreateICmpULT(x_loc, tile_width), in EmitPartialElementalTile()
2402 y_loc, x_loc, j); in EmitPartialElementalTile()
2458 llvm::Value* x_loc, int64 /*x_iter_num*/) { in EmitTileElementForCopy() argument
2464 {b_.getInt64(0), x_loc, y_loc}), in EmitTileElementForCopy()
2488 llvm::Value* x_loc, int64 /*x_iter_num*/) { in EmitTileElementForFusion() argument
2497 tiled_param_info->set_x(x_loc); in EmitTileElementForFusion()
[all …]
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dlayers_test.py3582 def _SpatialSoftmax(self, x_loc, y_loc, height, width, batch_size, nchannels): argument
3586 x_lin = np.expand_dims(np.array([height_lin[i] for i in x_loc]), 1)
3629 x_loc = [avg_x]
3632 np_keypoints = self._SpatialSoftmax(x_loc, y_loc, height, width, batch_size,
3652 x_loc, y_loc = zip(*edges)
3654 np_features[:, x_loc[c], y_loc[c], c] = 100.
3656 np_keypoints = self._SpatialSoftmax(x_loc, y_loc, height, width, batch_size,
3680 x_loc = [15, 2]
3683 np_features1[:, x_loc[c], y_loc[c], c] = 100.
3684 np_features2[:, x_loc[c], y_loc[c], c] = 100.
[all …]